package study.算法Algorithm.查找算法.插值查找;

import java.util.ArrayList;
import java.util.List;
/*
    插值查找注意事项：
        1、对于数据量较大，关键字分布比较均匀的查找表来说，采用插值查找，速度较快
        2、关键字分布不均匀的情况下，该方法不一定比二分查找要好
 */
public class InsertValueSearch {

    public static void main(String[] args) {

//        int arr[] = new int[100];
//        for (int i = 0; i < arr.length; i++) {
//            arr[i] = i;
//        }
        int arr[] = {1, 8, 10, 89, 89, 89, 1000, 1234};
        List resIndex = m01(arr, 0, arr.length - 1, 89);
        System.out.println("resIndex = " + resIndex);
    }

    /**
     * 插值查找算法    要求：数组是有序的
     *
     * @param arr     数组
     * @param left    左边索引
     * @param right   右边索引
     * @param findVal 查找值
     * @return 如果找到就返回集合，没找到返回null
     */
    private static List m01(int[] arr, int left, int right, int findVal) {

        //findVal < arr[left] || findVal > arr[right]  这两条件必须要，否则mid值可能越界
        if (left > right || findVal < arr[left] || findVal > arr[right]) {
            return null;
        }

        //求出mid   ，自适应的写法
        int mid = left + (right - left) * (findVal - arr[left]) / (arr[right] - arr[left]);
        int midVal = arr[mid];
        if (findVal>midVal){
            return m01(arr,mid+1,right,findVal);
        }else if (findVal<midVal){
            return m01(arr,left,mid+1,findVal);
        }else {
            List<Integer> list = new ArrayList<>();
            //向mid索引值的左边扫描，将所有满足查找值的下标加入到集合中
            int temp = mid - 1;
            while (true) {
                if (temp < 0 || arr[temp] != findVal) {
                    break;
                } else {
                    list.add(temp);
                    temp--;
                }
            }

            list.add(mid);

            //向mid索引值的右边扫描，将所有满足查找值的下标加入到集合中
            temp = mid + 1;
            while (true) {
                if (temp > right || arr[temp] != findVal) {
                    break;
                } else {
                    list.add(temp);
                    temp++;
                }
            }
            return list;
        }

    }
}
